package databases

import (
	"fmt"
	config "golang-blog/conf"
	"log"
	"time"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

// 使用Gorm 使用mysql

var (
	// MySql 数据库实例
	mySqlDB *gorm.DB
)

func initMysql() {
	var (
		host     = config.Viper().GetString("db.mysql.host")
		port     = config.Viper().GetString("db.mysql.port")
		username = config.Viper().GetString("db.mysql.username")
		password = config.Viper().GetString("db.mysql.password")
		dbname   = config.Viper().GetString("db.mysql.dbname")
	)

	// 连接语句：username:pass@tcp(ip:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local
	dsn := username + ":" + password + "@tcp(" + host + ":" + port + ")/" + dbname + "?charset=utf8mb4&parseTime=True&loc=Local"
	fmt.Println("dsn:", dsn)
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		panic(err)
	}

	db.Exec("alter database " + dbname + " character set utf8mb4")

	mySqlDB = db
	sqlDB, err := db.DB()
	if err != nil {
		log.Fatal("setupmusql connect database server failed.")
		return
	}
	// SetMaxIdleConns 设置空闲连接池中连接的最大数量
	sqlDB.SetMaxIdleConns(10)
	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(100)
	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(5 * time.Minute)

}

func GetMysqlDB() *gorm.DB {
	if mySqlDB == nil {
		initMysql()
		return mySqlDB
	}
	return mySqlDB
}
